"""
K折交叉验证全局配置
"""

# 全局配置
GLOBAL_CONFIG = {
    'k_folds': 8,
    'seed': 42,
    'data_root': 'data',
    'images_dir': 'data/images',
    'masks_dir': 'data/masks',
}

# 所有baseline模型列表
ALL_MODELS = [
    'unet',
    'deeplabv3plus',
    'fcn',
    'pspnet',
    'unetpp',
    'segnet'
]

# 模型配置映射
MODEL_CONFIGS = {
    'unet': {
        'module': 'baselines.unet',
        'class': 'UNet',
        'to_rgb': False,
    },
    'deeplabv3plus': {
        'module': 'baselines.deeplabv3plus',
        'class': 'DeepLabV3Plus',
        'to_rgb': True,
    },
    'fcn': {
        'module': 'baselines.fcn',
        'class': 'FCN',
        'to_rgb': True,
    },
    'pspnet': {
        'module': 'baselines.pspnet',
        'class': 'PSPNet',
        'to_rgb': True,
    },
    'unetpp': {
        'module': 'baselines.unetpp',
        'class': 'UNetPlusPlus',
        'to_rgb': False,
    },
    'segnet': {
        'module': 'baselines.segnet',
        'class': 'SegNet',
        'to_rgb': False,
    }
}
